"""
ComfyUI S4API Custom Node Package
Author: S4MUEL
GitHub: https://github.com/S4MUEL-404/ComfyUI-S4API
"""

# API Nodes
from .py.imageGemini import S4ImageWithGemini
from .py.imageGrok import S4ImageWithGrok
from .py.textDeepseek import S4TextWithDeepseek
from .py.textGemini import S4TextWithGemini
from .py.textOpenAI import S4TextWithOpenAI
from .py.textGrok import S4TextWithGrok

# Prompts Processing Nodes
from .py.promptsCombine import S4PromptsCombine
from .py.promptsInput import S4PromptsInput
from .py.promptsSplit import S4PromptsSplit
from .py.promptsReplace import S4PromptsReplace
from .py.promptsFromJanusPro import S4PromptsFromJanusPro

# Node class mappings
NODE_CLASS_MAPPINGS = {
    # API Nodes
    "💀S4ImageWithGemini": S4ImageWithGemini,
    "💀S4ImageWithGrok": S4ImageWithGrok,
    "💀S4TextWithDeepseek": S4TextWithDeepseek,
    "💀S4TextWithGemini": S4TextWithGemini,
    "💀S4TextWithOpenAI": S4TextWithOpenAI,
    "💀S4TextWithGrok": S4TextWithGrok,
    # Prompts Processing Nodes
    "💀S4PromptsCombine": S4PromptsCombine,
    "💀S4PromptsInput": S4PromptsInput,
    "💀S4PromptsSplit": S4PromptsSplit,
    "💀S4PromptsReplace": S4PromptsReplace,
    "💀S4PromptsFromJanusPro": S4PromptsFromJanusPro,
}

# Node display name mappings
NODE_DISPLAY_NAME_MAPPINGS = {
    # API Nodes
    "💀S4ImageWithGemini": "💀Image with Gemini",
    "💀S4ImageWithGrok": "💀Image with Grok",
    "💀S4TextWithDeepseek": "💀Text with Deepseek",
    "💀S4TextWithGemini": "💀Text with Gemini",
    "💀S4TextWithOpenAI": "💀Text with OpenAI",
    "💀S4TextWithGrok": "💀Text with Grok",
    # Prompts Processing Nodes
    "💀S4PromptsCombine": "💀Prompts Combine",
    "💀S4PromptsInput": "💀Prompts Input",
    "💀S4PromptsSplit": "💀Prompts Split",
    "💀S4PromptsReplace": "💀Prompts Replace",
    "💀S4PromptsFromJanusPro": "💀Prompts from Janus Pro",
}

__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
